"""
menu.py         菜单界面
util.py         工具 对用户的输入做检查. 对文件的读写
user.py         注册登录 
account.py      账户的操作
userrecord.txt  保存账户数据的文件
"""


from user import register
from user import login
from account import *

def start_menu():
    """
        构建开始菜单
        @param
        @return 
    """
    
    tips = '''
    *******************欢迎使用WoniuATM*******************
        ****************请选择操作菜单****************
    **************** 1. 注册 2.登录 3.退卡****************
            '''
    print(tips)
    option = input("请输入选择:")
    if option == "1":
        # NameError: name 'register' is not defined 名称叫做register是未定义的.
        register()
    elif option == "2":
        login()
    elif option == "3":
        # exit("退出时显示的消息") 这是系统自带的函数.用于立刻退出
        exit("谢谢,欢迎下次使用!")
    else:
        start_menu()
        

def main_menu():
    """
        构建登录成功后要显示的主菜单
    """
    tips = '''
    *********************欢迎登录蜗牛银行*********************
          ****************请选择操作菜单****************
    *****1.查询 2.取款 3.存款 4.转账 5.流水 6.返回 7.退卡*****
           '''
    print(tips)
    option = input("请输入选择:")
    if option == "1":
        query()
    elif option == "2":
        withdraw()
    elif option == "3":
        deposit()
    elif option == "4":
        transfer()
        pass    
    elif option == "5":
        pass    
    elif option == "6":
        pass    
    elif option == "7":
        exit("谢谢,欢迎下次使用!")
    else:
        main_menu()
    
# python的默认入口函数


if __name__ == "__main__":
    start_menu()









